package com.cardsys.controller;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.Random;

import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;

import com.cardsys.model.User;
import com.cardsys.service.UserService;

@Controller
public class PublicController extends BaseController {
	@Autowired
	private UserService userService;
	
	@RequestMapping(value = "/login", method = RequestMethod.POST) 
	public String handleLogin(String username, String password, 
			String randomCode, ModelMap modelMap, HttpSession session) {
		if (!randomCode.equalsIgnoreCase((String) session.getAttribute("rand"))) {
			modelMap.put("errors", "验证码输入错误!");
			return "login";
		}
		User user = userService.getUserByAccount(username);
		if (user != null && user.getPassword().equals(password)) {
			session.setAttribute("user", user);
			
			if ("admin".equals(user.getType())) {
				return "redirect:main.html";
			} else if ("teacher".equals(user.getType())) {
				return "redirect:tmain.html";
			} else {
				return "redirect:smain.html";
			}
		} else {
			modelMap.put("errors", "用户名或密码错误!");
			return "login";
		}
	}
	
	@RequestMapping(value = "/logout", method = RequestMethod.GET) 
	public String logout(HttpSession session) {
		session.removeAttribute("user");
		
		return "login";
	}
	
	@RequestMapping(value = "/code", method = RequestMethod.GET) 
	public void code(HttpServletRequest request, HttpServletResponse response) throws IOException {
		response.setHeader("Pragma","No-cache");     
        response.setHeader("Cache-Control","no-cache");     
        response.setDateHeader("Expires", 0);     
        //表明生成的响应是图片     
        response.setContentType("image/jpeg");     
             
        int width=100, height=18;     
        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);     
             
        Graphics g = image.getGraphics();     
        Random random = new Random();     
        g.setColor(getRandColor(200,250));     
        g.fillRect(1, 1, width-1, height-1);     
        g.setColor(new Color(102,102,102));     
        g.drawRect(0, 0, width-1, height-1);     
        g.setFont(new Font("Times New Roman", Font.PLAIN, 17));     
    
        g.setColor(getRandColor(160,200));     
    
        //画随机线     
        for (int i=0;i<155;i++)     
        {     
            int x = random.nextInt(width - 1);     
            int y = random.nextInt(height - 1);     
            int xl = random.nextInt(6) + 1;     
            int yl = random.nextInt(12) + 1;     
            g.drawLine(x,y,x + xl,y + yl);     
        }     
    
        //从另一方向画随机线     
        for (int i = 0;i < 70;i++)     
        {     
            int x = random.nextInt(width - 1);     
            int y = random.nextInt(height - 1);     
            int xl = random.nextInt(12) + 1;     
            int yl = random.nextInt(6) + 1;     
            g.drawLine(x,y,x - xl,y - yl);     
        }     
    
        //生成随机数,并将随机数字转换为字母     
        String sRand="";     
        for (int i=0;i<6;i++)     
        {     
            int itmp = random.nextInt(26) + 65;     
            char ctmp = (char)itmp;     
            sRand += String.valueOf(ctmp);     
            g.setColor(new Color(20+random.nextInt(110),20+random.nextInt(110),20+random.nextInt(110)));     
            g.drawString(String.valueOf(ctmp),15*i+10,16);     
        }     
    
        HttpSession session = request.getSession();     
        session.setAttribute("rand",sRand);     
        g.dispose();     
        ImageIO.write(image, "JPEG", response.getOutputStream()); 
	}
	
	private Color getRandColor(int fc,int bc)     
    {     
        Random random = new Random();     
        if(fc>255) fc=255;     
        if(bc>255) bc=255;     
        int r=fc+random.nextInt(bc-fc);     
        int g=fc+random.nextInt(bc-fc);     
        int b=fc+random.nextInt(bc-fc);     
        return new Color(r,g,b);     
    }  
}
